Skip to content

Conversation

@leisuzz
Copy link
Contributor

@leisuzz leisuzz commented Jan 15, 2025

What does this PR do?

Add vae tiling option to avoid OOM
Without vae tiling, memory will significantly increase and break the process during validation process, if the memory is not enough

Before submitting

Who can review?

Anyone in the community is free to review the PR once the tests have passed. Feel free to tag
members/contributors who may be interested in your PR.

@leisuzz
Copy link
Contributor Author

leisuzz commented Jan 15, 2025

@sayakpaul Please take a look at this function, thank you for your help!

@a-r-r-o-w a-r-r-o-w requested a review from sayakpaul January 15, 2025 09:08
Copy link
Member

@sayakpaul sayakpaul left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Nice!

@HuggingFaceDocBuilderDev

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.

@a-r-r-o-w a-r-r-o-w merged commit b0c8973 into huggingface:main Jan 15, 2025
9 checks passed
@leisuzz leisuzz deleted the tiling branch June 3, 2025 11:13
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants